from common.common_object import CommonObject
from common.utils import remove


class QueryModel(CommonObject):
    def __init__(self, trigger, query_str):
        self.trigger = trigger
        self.query_str = query_str.lstrip()
        self.query_arr = []
        self.init_query_arr()
        self.current_index = 0
        pass

    def init_query_arr(self, method=None):
        if method is None:
            # 默认使用空格分隔
            self.query_arr = remove(self.query_str.split(' '))
        else:
            # 允许用户自定义分隔方法
            self.query_arr = method(self.query_str)
        pass

    def add_index(self, default_value=1):
        self.current_index += default_value
        return self

    pass
